package com.sheng.leetcode.year2022.month04.day19;

import org.junit.Test;

/**
 * @author liusheng
 * @date 2022/04/19
 *
 * 821. 字符的最短距离
 *
 * 给你一个字符串 s 和一个字符 c ，且 c 是 s 中出现过的字符。
 *
 * 返回一个整数数组 answer ，其中 answer.length == s.length 且 answer[i] 是 s 中从下标 i 到离它 最近 的字符 c 的 距离 。
 *
 * 两个下标 i 和 j 之间的 距离 为 abs(i - j) ，其中 abs 是绝对值函数。
 *
 * 示例 1：
 *
 * 输入：s = "loveleetcode", c = "e"
 * 输出：[3,2,1,0,1,0,0,1,2,2,1,0]
 * 解释：字符 'e' 出现在下标 3、5、6 和 11 处（下标从 0 开始计数）。
 * 距下标 0 最近的 'e' 出现在下标 3 ，所以距离为 abs(0 - 3) = 3 。
 * 距下标 1 最近的 'e' 出现在下标 3 ，所以距离为 abs(1 - 3) = 2 。
 * 对于下标 4 ，出现在下标 3 和下标 5 处的 'e' 都离它最近，但距离是一样的 abs(4 - 3) == abs(4 - 5) = 1 。
 * 距下标 8 最近的 'e' 出现在下标 6 ，所以距离为 abs(8 - 6) = 2 。
 * 示例 2：
 *
 * 输入：s = "aaab", c = "b"
 * 输出：[3,2,1,0]
 *
 * 提示：
 * 1 <= s.length <= 104
 * s[i] 和 c 均为小写英文字母
 * 题目数据保证 c 在 s 中至少出现一次
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/shortest-distance-to-a-character
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class LeetCode0821 {

    @Test
    public void test01(){
//        String s = "loveleetcode";
//        char c = 'e';
        String s = "aaab";
        char c = 'b';
        int[] ints = new Solution().shortestToChar(s, c);
        for (int i = 0; i < ints.length; i++) {
            if (i == 0) {
                System.out.print("[" + ints[i] + ", ");
            } else if (i == ints.length - 1) {
                System.out.print(ints[i] + "]");
            } else {
                System.out.print(ints[i] + ", ");
            }
        }
    }
}
class Solution {
    public int[] shortestToChar(String s, char c) {
        int[] answer = new int[s.length()];
        char[] chars = s.toCharArray();
        for (int i = 0; i < s.length(); i++) {
            int left = 0;
            int right = 0;
            if (chars[i] == c) {
                answer[i] = 0;
                continue;
            }
            //先求这个坐标左边e的位置离现在的距离
            String s1 = s.substring(0, i);
            int i1 = s1.lastIndexOf(c);
            if (i1 != -1) {
                left = i - i1;
            }
            //再求这个坐标右边e的位置离现在的距离
            String s2 = s.substring(i, s.length());
            int i2 = s2.indexOf(c);
            if (i2 != -1) {
                right = i2;
            }
            if (left == 0) {
                answer[i] = right;
            } else if (right == 0) {
                answer[i] = left;
            } else {
                if (left <= right) {
                    answer[i] = left;
                } else {
                    answer[i] = right;
                }
            }
        }
        return answer;
    }
}
